// Copyright (c) Microsoft
// All rights reserved
// Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License.
// You may obtain a copy of the License at http://www.apache.org/licenses/LICENSE-2.0
// THIS CODE IS PROVIDED *AS IS* BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED,
// INCLUDING WITHOUT LIMITATION ANY IMPLIED WARRANTIES OR CONDITIONS OF TITLE, FITNESS FOR A PARTICULAR PURPOSE, MERCHANTABLITY OR NON-INFRINGEMENT.
// See the Apache Version 2.0 License for specific language governing permissions and limitations under the License.
/// <tags>P1</tags>
/// <summary>Verify availablity of array.get_extent() properties for 1D, 2D and 3D</summary>
// RUN: %amp_device -D__GPU__ %s -m32 -emit-llvm -c -S -O3 -o %t.ll && mkdir -p %t
// RUN: %llc -march=c -o %t/kernel_.cl < %t.ll
// RUN: cat %opencl_math_dir/opencl_math.cl %t/kernel_.cl > %t/kernel.cl
// RUN: pushd %t && objcopy -B i386:x86-64 -I binary -O elf64-x86-64 kernel.cl %t/kernel.o && popd
// RUN: %cxxamp %link %t/kernel.o %s -o %t.out && %t.out
#include "./../../../member.h"

bool test_feature_1d(int *edata)
{
    const int rank = 1;
    extent<rank> e1(edata);
    array<int, rank> src(e1);

    if (src.get_extent()[0] != edata[0])
        return false;

    return true;
}

bool test_feature_2d(int *edata)
{
    const int rank = 2;
    extent<rank> e1(edata);
    array<int, rank> src(e1);

    if (src.get_extent()[0] != edata[0])
        return false;

    if (src.get_extent()[1] != edata[1])
        return false;

    return true;
}

bool test_feature_3d(int *edata)
{
    const int rank = 3;
    extent<rank> e1(edata);
    array<int, rank> src(e1);

    if (src.get_extent()[0] != edata[0])
        return false;

    if (src.get_extent()[1] != edata[1])
        return false;

    if (src.get_extent()[2] != edata[2])
        return false;

    return true;
}

#define MAX 5
int main() 
{
    int edata[MAX];
    for (int i = 0; i < MAX; i++)
        edata[i] = i+1;

    int passed = 
        test_feature_1d(edata) && test_feature_2d(edata) && test_feature_3d(edata)
            ? 0 : 1;

    return passed;
}
